import matplotlib.pyplot as plt

# 用于存储radius_w的值
radius_w_data = []
# 用于存储AoI的值
AoI_data = []
# 打开xlsx文件，读取数据，第一列数据为radius_w，第二列数据为AoI
import openpyxl

# 打开xlsx文件
workbook = openpyxl.load_workbook("./radiusVSAoI.xlsx")

# 选择第一个工作表
worksheet = workbook.worksheets[0]
# 遍历工作表中的每一行数据
for row in worksheet.iter_rows(min_row=2, values_only=True):
    # 将第一列数据添加到radius_w_data列表中
    radius_w_data.append(row[0])
    # 将第二列数据添加到AoI_data列表中
    AoI_data.append(row[1])

# 打印数据
print(radius_w_data)
print(AoI_data)

# 获取radius_w_data的最后一个元素
px = radius_w_data[-1]
# 获取AoI_data的最后一个元素
py = AoI_data[-1]
# 绘制散点图
plt.scatter(radius_w_data, AoI_data)
# 绘制折线图
plt.plot(radius_w_data, AoI_data)


plt.scatter(radius_w_data, [py] * len(radius_w_data))
plt.plot(radius_w_data, [py] * len(radius_w_data))


# 添加标题和坐标轴标签
plt.title("Scatter Plot of radius_w vs AoI")
plt.xlabel("radius_w")
plt.ylabel("AoI")

# 显示图形
plt.show()
